package com.founder.bdyx.core.pdfUtil;

import com.founder.bdyx.modules.sys.model.ApplyInfo;
import com.itextpdf.text.*;
import com.itextpdf.text.pdf.*;
import com.itextpdf.text.pdf.draw.DottedLineSeparator;
import com.itextpdf.text.pdf.draw.LineSeparator;

import java.io.*;
import java.util.*;
import java.util.List;

import com.itextpdf.text.pdf.AcroFields;
import com.itextpdf.text.pdf.AcroFields.Item;
import net.sf.json.JSONArray;
import net.sf.json.JSONObject;

public class CreateTemplate {

    // 定义全局的字体静态变量
    private static Font titlefont;
    private static Font headfont;
    private static Font keyfont;
    private static Font textfont;
    // 最大宽度
    private static int maxWidth = 520;
    // 静态代码块
    static {
        try {
            // 不同字体（这里定义为同一种字体：包含不同字号、不同style）
            BaseFont bfChinese = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
            titlefont = new Font(bfChinese, 16, Font.BOLD);
            headfont = new Font(bfChinese, 14, Font.BOLD);
            keyfont = new Font(bfChinese, 10, Font.BOLD);
            textfont = new Font(bfChinese, 10, Font.NORMAL);

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 创建单元格(指定字体)
     * @param value
     * @param font
     * @return
     */
    public PdfPCell createCell(String value, Font font) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(Element.ALIGN_CENTER);
        cell.setPhrase(new Phrase(value, font));
        return cell;
    }
    /**
     * 创建单元格（指定字体、水平..）
     * @param value
     * @param font
     * @param align
     * @return
     */
    public PdfPCell createCell(String value, Font font, int align) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        cell.setPhrase(new Phrase(value, font));
        return cell;
    }
    /**
     * 创建单元格（指定字体、水平居..、单元格跨x列合并）
     * @param value
     * @param font
     * @param align
     * @param colspan
     * @return
     */
    public PdfPCell createCell(String value, Font font, int align, int colspan) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        cell.setColspan(colspan);
        cell.setPhrase(new Phrase(value, font));
        return cell;
    }
    /**
     * 创建单元格（指定字体、水平居..、单元格跨x列合并、设置单元格内边距）
     * @param value
     * @param font
     * @param align
     * @param colspan
     * @param boderFlag
     * @return
     */
    public PdfPCell createCell(String value, Font font, int align, int colspan, boolean boderFlag) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        cell.setColspan(colspan);
        cell.setPhrase(new Phrase(value, font));
        cell.setPadding(3.0f);
        if (!boderFlag) {
            cell.setBorder(0);
            cell.setPaddingTop(15.0f);
            cell.setPaddingBottom(8.0f);
        } else if (boderFlag) {
            cell.setBorder(0);
            cell.setPaddingTop(0.0f);
            cell.setPaddingBottom(15.0f);
        }
        return cell;
    }
    /**
     * 创建单元格（指定字体、水平..、边框宽度：0表示无边框、内边距）
     * @param value
     * @param font
     * @param align
     * @param borderWidth
     * @param paddingSize
     * @param flag
     * @return
     */
    public PdfPCell createCell(String value, Font font, int align, float[] borderWidth, float[] paddingSize, boolean flag) {
        PdfPCell cell = new PdfPCell();
        cell.setVerticalAlignment(Element.ALIGN_MIDDLE);
        cell.setHorizontalAlignment(align);
        cell.setPhrase(new Phrase(value, font));
        cell.setBorderWidthLeft(borderWidth[0]);
        cell.setBorderWidthRight(borderWidth[1]);
        cell.setBorderWidthTop(borderWidth[2]);
        cell.setBorderWidthBottom(borderWidth[3]);
        cell.setPaddingTop(paddingSize[0]);
        cell.setPaddingBottom(paddingSize[1]);
        if (flag) {
            cell.setColspan(2);
        }
        return cell;
    }

    /**
     * 创建默认列宽，指定列数、水平(居中、右、左)的表格
     * @param colNumber
     * @param align
     * @return
     */
    public PdfPTable createTable(int colNumber, int align) {
        PdfPTable table = new PdfPTable(colNumber);
        try {
            table.setTotalWidth(maxWidth);
            table.setLockedWidth(true);
            table.setHorizontalAlignment(align);
            table.getDefaultCell().setBorder(1);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return table;
    }
    /**
     * 创建指定列宽、列数的表格
     * @param widths
     * @return
     */
    public PdfPTable createTable(float[] widths) {
        PdfPTable table = new PdfPTable(widths);
        try {
            table.setTotalWidth(maxWidth);
            table.setLockedWidth(true);
            table.setHorizontalAlignment(Element.ALIGN_CENTER);
            table.getDefaultCell().setBorder(1);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return table;
    }
    /**
     * 创建空白的表格
     * @return
     */
    public PdfPTable createBlankTable() {
        PdfPTable table = new PdfPTable(1);
        table.getDefaultCell().setBorder(0);
        table.addCell(createCell("", keyfont));
        table.setSpacingAfter(20.0f);
        table.setSpacingBefore(20.0f);
        return table;
    }

    /**
     * 打印执行单模板
     * @param document
     * @throws Exception
     */
    public void execSheetTemplate(Document document) throws Exception {
        Paragraph p = new Paragraph("申请单号：210416000001",textfont);
        document.add(p);
    }

    public static void execOrderTemplate(ApplyInfo applyInfo, File file) throws Exception {
        Document document = new Document(PageSize.A4);
        PdfWriter writer = PdfWriter.getInstance(document, new FileOutputStream(file));
        Paragraph p = new Paragraph("申请单号：" + applyInfo.getUuid(),textfont);
        document.open();
        document.add(p);
        document.close();
        writer.close();

    }

    // 生成PDF文件
    public void generatePDF(Document document) throws Exception {
        // 段落
        Paragraph paragraph = new Paragraph("美好的一天从早起开始！", titlefont);
        paragraph.setAlignment(1); //设置文字居中 0靠左   1，居中     2，靠右
        paragraph.setIndentationLeft(12); //设置左缩进
        paragraph.setIndentationRight(12); //设置右缩进
        paragraph.setFirstLineIndent(24); //设置首行缩进
        paragraph.setLeading(20f); //行间距
        paragraph.setSpacingBefore(5f); //设置段落上空白
        paragraph.setSpacingAfter(10f); //设置段落下空白

        // 直线
        Paragraph p1 = new Paragraph();
        p1.add(new Chunk(new LineSeparator()));

        // 点线
        Paragraph p2 = new Paragraph();
        p2.add(new Chunk(new DottedLineSeparator()));

        // 超链接
        Anchor anchor = new Anchor("baidu");
        anchor.setReference("www.baidu.com");

        // 定位
        Anchor gotoP = new Anchor("goto");
        gotoP.setReference("#top");

//        // 添加图片
//        Image image = Image.getInstance("https://img-blog.csdn.net/20180801174617455?watermark/2/text/aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNzg0ODcxMA==/font/5a6L5L2T/fontsize/400/fill/I0JBQkFCMA==/dissolve/70");
//        image.setAlignment(Image.ALIGN_CENTER);
//        image.scalePercent(40); //依照比例缩放

        // 表格
        PdfPTable table = createTable(new float[] { 40, 120, 120, 120, 80, 80 });
        table.addCell(createCell("美好的一天", headfont, Element.ALIGN_LEFT, 6, false));
        table.addCell(createCell("早上9:00", keyfont, Element.ALIGN_CENTER));
        table.addCell(createCell("中午11:00", keyfont, Element.ALIGN_CENTER));
        table.addCell(createCell("中午13:00", keyfont, Element.ALIGN_CENTER));
        table.addCell(createCell("下午15:00", keyfont, Element.ALIGN_CENTER));
        table.addCell(createCell("下午17:00", keyfont, Element.ALIGN_CENTER));
        table.addCell(createCell("晚上19:00", keyfont, Element.ALIGN_CENTER));
        Integer totalQuantity = 0;
        for (int i = 0; i < 5; i++) {
            table.addCell(createCell("起床", textfont));
            table.addCell(createCell("吃午饭", textfont));
            table.addCell(createCell("午休", textfont));
            table.addCell(createCell("下午茶", textfont));
            table.addCell(createCell("回家", textfont));
            table.addCell(createCell("吃晚饭", textfont));
            totalQuantity ++;
        }
        table.addCell(createCell("总计", keyfont));
        table.addCell(createCell("", textfont));
        table.addCell(createCell("", textfont));
        table.addCell(createCell("", textfont));
        table.addCell(createCell(String.valueOf(totalQuantity) + "件事", textfont));
        table.addCell(createCell("", textfont));

        document.add(paragraph);
        document.add(anchor);
        document.add(p2);
        document.add(gotoP);
        document.add(p1);
        document.add(table);
//        document.add(image);
    }

    /**
     * PDF文件合并 使用PdfCopy
     *
     * @author
     * @param files
     * @param newfile
     */
    public static boolean mergePdfFiles(List<File> files, String newfile) {
        boolean retValue = false;
        Document document = null;
        try {
            document = new Document();
            PdfCopy copy = new PdfCopy(document, new FileOutputStream(newfile));
            document.open();

            for (int i = 0; i < files.size(); i++) {// 几个pdf文件循环
                PdfReader reader = new PdfReader(files.get(i).getCanonicalPath());
                int n = reader.getNumberOfPages();
                for (int j = 1; j <= n; j++) {// 一个文件有多少页循环
                    document.newPage();
                    PdfImportedPage page = copy.getImportedPage(reader, j);
                    copy.addPage(page);
                }
                reader.close();
            }
            retValue = true;
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            document.close();
        }
        return retValue;
    }

    /**
     * 合并PDF  PdfWriter实现方式
     *
     * @author ShaoMin
     * @throws Exception
     *
     */
    public void mergePdf() throws Exception {

        String[] files = { "source/pdf/1.pdf", "source/pdf/2.pdf" };
        String savepath = "source/pdf/mergePdf.pdf";

        Document document = new Document();

        PdfWriter writer = PdfWriter.getInstance(document, new FileOutputStream(savepath));
        // PdfCopy copy = new PdfCopy(document, new
        // FileOutputStream(newfile));//使用copy
        document.open();
        PdfContentByte cb = writer.getDirectContent();// 得到层

        for (int i = 0; i < files.length; i++) {
            PdfReader reader = new PdfReader(files[i]);
            int n = reader.getNumberOfPages();
            for (int j = 1; j <= n; j++) {
                document.newPage();
                // PdfImportedPage page = copy.getImportedPage(reader, j);
                // copy.addPage(page);
                PdfImportedPage page = writer.getImportedPage(reader, j);
                cb.addTemplate(page, 0, 0);// 使用writer需要使用pdf的层,然后后添加
            }
        }
        document.close();

        // 使用PdfCopy 实现Pdf合并
//      mergePdfFiles(files, savepath);
    }

    /**
     * 根据模板生成pdf
     */
    public void createByTemplate() throws IOException, DocumentException {
        String tPdfTemplateFile = "d:/pdf/execOrderTemplate.pdf";// 获取模板路径
        String tPdfResultFile = "d:/pdf/201.pdf";// 生成的文件路径

        // 2-解析PDF模板
        FileOutputStream fos = new FileOutputStream(tPdfResultFile);// 需要生成PDF
        PdfReader reader = new PdfReader(tPdfTemplateFile);// 模板
        PdfStamper mPdfStamper = new PdfStamper(reader, fos);// 解析
        // 3-获取到模板上预定义的参数域
        AcroFields form = mPdfStamper.getAcroFields();
        // 获取模板中定义的变量
        Map<String, Item> acroFieldMap = form.getFields();

        // 循环解析模板定义的表单域
        int len = 4;
        for (Map.Entry<String, Item> entry : acroFieldMap.entrySet()) {
            // 获得块名
            String fieldName = entry.getKey();
            String fieldValue = "fill_" + len;
            System.out.println(fieldName + ":" + fieldValue);
            form.setField(fieldName, fieldValue);
            len++;
        }

        // 模板中的变量赋值之后不能编辑
        mPdfStamper.setFormFlattening(true);
        reader.close();// 阅读器关闭,解析器暂时不关闭，因为创建动态表格还需要使用
        mPdfStamper.close();
    }

    /**
     * 获取模板表单，并赋值 固定用法
     *
     * @author ShaoMin
     * @throws Exception
     */
    public static void fillFormDatas() throws Exception {

        // 1-封装的数据，这边的key与pdf模板的域名保持一致
        Map<String, String> mMapDatas = new HashMap<String, String>();
        mMapDatas.put("uuid", "210507000001");// 客户姓名
        mMapDatas.put("isPay", "已支付");// 合同号
        mMapDatas.put("transType", "邮寄");// 保单个数
        mMapDatas.put("printName", "二愣子");// 保全类型
        mMapDatas.put("printTime", "2014-11-1");// 保全失算金额
        mMapDatas.put("idcName", "aaa");// 受理人
        mMapDatas.put("startTime", "2014-11-1");
        mMapDatas.put("endTime", "2014-11-1");
        mMapDatas.put("idcNo", "2014-11-1");
        mMapDatas.put("depName", "2014-11-1");
        mMapDatas.put("cureNo", "2014-11-1");
        mMapDatas.put("purpContent", "2014-11-1");
        mMapDatas.put("printCount", "2014-11-1");
        mMapDatas.put("singlePrice", "2014-11-1");
        mMapDatas.put("transPrice", "2014-11-1");
        mMapDatas.put("payPrice", "2014-11-1");
        mMapDatas.put("recvName", "2014-11-1");
        mMapDatas.put("recvMobile", "2014-11-1");
        mMapDatas.put("recvAddr", "2014-11-1");


        // 2-模板和生成的pdf
        Random a = new Random();
        a.nextInt();
        String tPdfTemplateFile = "d:/pdf/execOrderTemplate.pdf";// 获取模板路径
        String tPdfResultFile = "d:/pdf/201.pdf";// 生成的文件路径

        // 3-解析PDF模板
        FileOutputStream fos = new FileOutputStream(tPdfResultFile);// 需要生成PDF
        PdfReader reader = new PdfReader(tPdfTemplateFile);// 模板
        PdfStamper mPdfStamper = new PdfStamper(reader, fos);// 解析

        // 4-获取到模板上预定义的参数域
        AcroFields form = mPdfStamper.getAcroFields();
        // 获取模板中定义的变量
        Map<String, Item> acroFieldMap = form.getFields();
        // 循环解析模板定义的表单域
        for (Map.Entry<String, Item> entry : acroFieldMap.entrySet()) {
            // 获得块名
            String fieldName = entry.getKey();
            String fieldValue = mMapDatas.get(fieldName);// 通过名字，获取传入的参数值
            if (!"".equals(fieldValue)) {
                // 为模板中的变量赋值(key与pdf模板定义的域名一致)
                form.setField(fieldName, fieldValue);
                System.out.println(fieldName + "," + fieldValue);
            }
        }

        // 模板中的变量赋值之后不能编辑
        mPdfStamper.setFormFlattening(true);
        reader.close();// 阅读器关闭,解析器暂时不关闭，因为创建动态表格还需要使用
//        mPdfStamper.close();
    }

    /**
     * 给PDF表格赋值 值动态的，一般建议使用模板， 直接创建绝对位置的表格
     *
     * @author ShaoMin
     * @throws Exception
     */
    public static void fillTableDatas() throws Exception {

        // 1-模板和生成的pdf

        String tPdfTemplateFile = "d:/pdf/execOrderTemplate.pdf";// 获取模板路径
        String tPdfResultFile = "d:/pdf/201222.pdf";// 生成的文件路径

        // 2-解析PDF模板
        FileOutputStream fos = new FileOutputStream(tPdfResultFile);// 需要生成PDF
        PdfReader reader = new PdfReader(tPdfTemplateFile);// 模板
        PdfStamper mPdfStamper = new PdfStamper(reader, fos);// 解析

        // 3-获取到模板上预定义的参数域
        AcroFields form = mPdfStamper.getAcroFields();
        // 获取模板中定义的变量
        Map<String, Item> acroFieldMap = form.getFields();
        // 循环解析模板定义的表单域
        int len = 4;
        for (Map.Entry<String, Item> entry : acroFieldMap.entrySet()) {
            // 获得块名
            String fieldName = entry.getKey();
            String fieldValue = "fill_" + len;
            System.out.println(fieldName + ":" + fieldValue);
            form.setField(fieldName, fieldValue);
            len++;
        }

        // 模板中的变量赋值之后不能编辑
        mPdfStamper.setFormFlattening(true);
        reader.close();// 阅读器关闭,解析器暂时不关闭，因为创建动态表格还需要使用
//        mPdfStamper.close();

    }

    /**
     * pdf导出
     * @param datas key-value对象json数组，一个数组元素为一份（一页）
     * @param templatePath 模板绝对路径
     * @param newPDF 导出pdf绝对路径
     * @throws Exception
     */
    public static void pdfout(JSONArray datas, String templatePath, File newPDF) throws Exception {
        PdfReader reader;
        FileOutputStream out;
        ByteArrayOutputStream bos;
        PdfStamper stamper;
        List<PdfReader> list = new ArrayList<PdfReader>();

        //try {c://windows//fonts//simsun.ttc
        /*
         * ===========================================
         * 模版和填充内容中的字体必须都是黑体！！！否则乱码！！！！！！！！
         * ===========================================
         */
        //BaseFont bf = BaseFont.createFont("d:/pdf/simhei.ttf" , BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);
        BaseFont bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
        out = new FileOutputStream(newPDF);// 输出流
        //文字类的内容处理
        for (int i = 0; i < datas.size(); i++) {

            JSONObject object = JSONObject.fromObject(datas.get(i));
            PdfReader pdfReader = null;
            reader = new PdfReader(templatePath);// 读取pdf模板
            bos = new ByteArrayOutputStream();
            stamper = new PdfStamper(reader, bos);
            AcroFields form = stamper.getAcroFields();
            form.addSubstitutionFont(bf);
            Iterator<String> it = object.keys();
            while (it.hasNext()) {
                // 获得key
                String key = it.next();
                String value = object.getString(key);
                value = parseNullStr(value);
                if (key.equals("imagepath")) {
                    // 书签名
                    String fieldName = key;
                    // 图片路径
                    String imagePath2 = value;
                    // 通过域名获取所在页和坐标，左下角为起点
                    List<com.itextpdf.text.pdf.AcroFields.FieldPosition> lst = form.getFieldPositions(fieldName);
                    for (int j = 0; j < lst.size(); j++) {
                        com.itextpdf.text.pdf.AcroFields.FieldPosition fied = lst.get(j);
                        int pageNo = fied.page;
                        Rectangle signRect = fied.position;
                        float x = signRect.getLeft();
                        float y = signRect.getBottom();
                        // 读图片
                        Image image = Image.getInstance(imagePath2);
                        // 获取操作的页面
                        PdfContentByte under = stamper.getOverContent(pageNo);
                        // 根据域的大小缩放图片
                        image.scaleToFit(signRect.getWidth(), signRect.getHeight());
                        // 添加图片
                        image.setAbsolutePosition(x, y);
                        under.addImage(image);
                    }
                } else {
                    form.setFieldProperty(key, "textfont", bf, null);
                    form.setField(key, value);
                }
            }

            stamper.setFormFlattening(true);// 如果为false，生成的PDF文件可以编辑，如果为true，生成的PDF文件不可以编辑
            stamper.close();
            pdfReader = new PdfReader(bos.toByteArray());
            list.add(pdfReader);
        }

        Document doc = new Document();
        PdfCopy copy = new PdfCopy(doc, out);
        doc.open();
        for (int k = 0; k < list.size(); k++) {
            PdfReader pdfReader = list.get(k);
            PdfImportedPage importPage = copy.getImportedPage(pdfReader, 1);
            copy.addPage(importPage);
        }
        doc.close();

    }

    public static String parseNullStr(String str){
        if(str == null){
            return "";
        }
        return str;
    }
}
